package com.stud.common.core.mybatis;

import com.baomidou.mybatisplus.extension.service.IService;
import com.baomidou.mybatisplus.extension.toolkit.SqlHelper;
import com.stud.common.core.context.SecurityContextHolder;
import com.stud.common.core.web.domain.BaseEntity;

import java.io.Serializable;
import java.util.Arrays;
import java.util.Date;
import java.util.List;
import java.util.Objects;

/**
 * @author Stud
 * @title mybatis-plus二次封装父类
 * @desription 用于扩展
 * @data 2022-08-20 10:53
 */
public interface BaseService<T extends BaseEntity> extends IService<T> {

    /**
     * 根据 ID 选择修改
     *
     * @param entity 实体对象
     */
    default boolean update(T entity) {
        if (Objects.isNull(entity.getUpdateTime())) {
            entity.setUpdateTime(new Date());
        }
        if (Objects.isNull(entity.getUpdateBy())) {
            entity.setUpdateBy(String.valueOf(SecurityContextHolder.getUserId()));
        }
        return updateById(entity);
    }

    /**
     * 根据 ID 删除
     *
     * @param ids 主键ID数组
     */
    default boolean removeByIds(Serializable[] ids) {
        return removeByIds(Arrays.asList(ids));
    }

    /**
     * 插入一条记录（选择字段，策略插入）
     *
     * @param entity 实体对象
     */
    @Override
    default boolean save(T entity) {

        if (Objects.isNull(entity.getCreateTime())) {
            entity.setCreateTime(new Date());
        }
        if (Objects.isNull(entity.getCreateBy())) {
            entity.setCreateBy(String.valueOf(SecurityContextHolder.getUserId()));
        }

        return SqlHelper.retBool(getBaseMapper().insert(entity));
    }

    /**
     * 高效批量插入
     *
     * @param entityList 实体对象
     */
    Integer insertBatchFast(List<T> entityList);

    /**
     * 根据 ID 修改,属性为空则赋值为空
     *
     * @param entity 实体对象
     */
    boolean updateFullById(T entity);
}
